Create new column using custom function pandas df error